package com.yufang.Day09_29;

/**
 * 66
 * 给定一个由整数组成的非空数组所表示的非负整数，在该数的基础上加一。
 * 最高位数字存放在数组的首位， 数组中每个元素只存储单个数字。
 * 你可以假设除了整数 0 之外，这个整数不会以零开头。
 * 示例 1:
 * 输入: [1,2,3]
 * 输出: [1,2,4]
 * 解释: 输入数组表示数字 123。
 * 示例 2:
 * 输入: [4,3,2,1]
 * 输出: [4,3,2,2]
 * 解释: 输入数组表示数字 4321。
 */
public class D_2 {

    public static void main(String[] args) {

        int[] nums = new int[5];
        nums[0] = 9;
        nums[1] = 8;
        nums[2] = 9;
        nums[3] = 8;
        nums[4] = 9;
        int[] ans = getAns(nums);
        for (int an : ans) {
            System.out.print(an+"  ");
        }

    }

    private static int[] getAns(int[] digits) {

        int[] ans = new int[digits.length];
        int temp = digits[digits.length-1] + 1;
        for (int i = digits.length-1; i >-1 ; i--) {
            if(temp == 10){
                ans[i] = 0;
                //进位
                if(i-1 != -1){
                    temp = digits[i-1] += 1;
                }

            }else {
                ans[i] = temp;
                if(i-1 != -1){
                    temp = digits[i-1];
                }
            }
        }
        if(ans[0] == 0){
            ans = new int[digits.length+1];
            ans[0] = 1;
            for (int i = 1; i < ans.length; i++) {
                ans[i] = 0;
            }
        }
        return ans;

    }

}
